from deepface import DeepFace
import numpy as np

if __name__ == "__main__":
    models = ["VGG-Face", "Facenet", "OpenFace", "DeepFace", "ArcFace"]
    objs = DeepFace.analyze(img_path="image_of_0.png", actions=['age', 'gender', 'race', 'emotion'])
    print(objs)
    embedding_objs = DeepFace.represent(img_path="E:/data/test/skin_test_20231124104715.jpg", model_name=models[0])
    embedding = embedding_objs[0]["embedding"]
    embedding = np.array(embedding)
    print(embedding)
